Skip to content

Conversation

@constantinpape
Copy link
Contributor

No description provided.

@constantinpape
Copy link
Contributor Author

Requires constantinpape/torch-em#529

@constantinpape constantinpape marked this pull request as ready for review May 19, 2025 09:35
@constantinpape
Copy link
Contributor Author

Hi @schilling40 ,
this PR implements the domain adaptation / semi-supervised training functionality.
The main function is flamingo_tools.training.mean_teacher_training.mean_teacher_training, which supports supervised domain adaptation, unsupervised domain adaptation and semi-supervised learning (details are explained in the function signature). The script scripts/training/sgn_domain_adaptation.py shows how to use it for unsupervised domain adaptation.

In addition, I refactored some training functionality, so this also includes some minor changes to scripts/training/train_distance_unet.py. (And I actually uncovered a minor bug along the way, but it doesn't affect prev. trainings). The changes should be fine, but I haven't checked that it actually still works.

@schilling40 schilling40 merged commit 534ab38 into master May 28, 2025
6 checks passed
@schilling40 schilling40 deleted the domain-adaptation branch May 28, 2025 12:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants